package com.atwulidun.tree;

public class ArrBinaryTreeDemo {
    public static void main(String[] args) {
        // 一、创建一个顺序存储二叉树数组
        int[] arr = {1,2,3,4,5,6,7};
        // 二、创建一棵顺序存储二叉树
        ArrBinaryTree arrBinaryTree = new ArrBinaryTree(arr);
        // 三、遍历
        // 1.前序遍历
        System.out.println("前序遍历");
        arrBinaryTree.preOrder();//1 2 4 5 3 6 7
        // 2.中序遍历
        System.out.println("中序遍历");
        arrBinaryTree.infixOrder();//4 2 5 1 6 3 7
        // 3.后序遍历
        System.out.println("后序遍历");
        arrBinaryTree.postOrder();//4 5 2 6 7 3 1
    }
}

// 创建一个顺序存储二叉树类
class ArrBinaryTree {
    // 定义一个存储数据结点的数组
    int[] arr;

    public ArrBinaryTree(int[] arr) {
        this.arr = arr;
    }

    // 以下三个为遍历方法
    // 前序遍历
    public void preOrder(int index) {
        // 首先对arr进行判断
        if (arr == null || arr.length == 0) {
            System.out.println("顺序存储二叉树为空，不能前序遍历！");
            return;
        }
        // 先输出当前结点
        System.out.print(arr[index] + " ");
        // 再递归输出左子树
        // 注意判断左结点的坐标，不能越界
        if (2 * index + 1 < arr.length) {
            preOrder(2 * index + 1);
        }
        // 再递归输出右子树
        if (2 * index + 2 < arr.length) {
            preOrder(2 * index + 2);
        }
    }

    // 中序遍历
    public void infixOrder(int index) {
        if (arr == null || arr.length == 0) {
            System.out.println("顺序存储二叉树为空，不能中序遍历！");
            return;
        }
        // 先递归输出左子树
        if (2 * index + 1 < arr.length) {
            infixOrder(2 * index + 1);
        }
        // 再输出当前结点
        System.out.print(arr[index] + " ");
        // 再递归输出右子树
        if (2 * index + 2 < arr.length) {
            infixOrder(2 * index + 2);
        }
    }

    // 后序遍历
    public void postOrder(int index) {
        if (arr == null || arr.length == 0) {
            System.out.println("顺序存储二叉树为空，不能后序遍历！");
            return;
        }
        // 先递归输出左子树
        if (2 * index + 1 < arr.length) {
            postOrder(2 * index + 1);
        }
        // 再递归输出右子树
        if (2 * index + 2 < arr.length) {
            postOrder(2 * index + 2);
        }
        // 最后再输出当前结点
        System.out.print(arr[index] + " ");
    }

    // 重载三个遍历方法
    // 前序遍历
    public void preOrder() {
        preOrder(0);
        System.out.println();
    }

    // 中序遍历
    public void infixOrder() {
        infixOrder(0);
        System.out.println();
    }

    // 后序遍历
    public void postOrder() {
        postOrder(0);
        System.out.println();
    }
}